class Student:
    def __play(self):
        print("我在玩")
    def homework(self,flag):
        if flag =="YES":
            self.__play()
        else:
            print("我在写作业")
s1=Student()
flag = input("你写完作业了吗?YES/NO\n")
s1.homework(flag)